package com.iflytek;

import android.annotation.SuppressLint;
import android.content.Context;
import android.content.SharedPreferences;
import android.os.Bundle;
import android.os.Environment;
import android.os.Handler;
import android.os.Message;
import android.widget.Toast;

import com.iflytek.cloud.ErrorCode;
import com.iflytek.cloud.RecognizerListener;
import com.iflytek.cloud.RecognizerResult;
import com.iflytek.cloud.SpeechConstant;
import com.iflytek.cloud.SpeechError;
import com.iflytek.cloud.SpeechRecognizer;
import com.iflytek.cloud.SpeechUtility;
import com.iflytek.speech.util.FucUtil;
import com.iflytek.speech.util.JsonParser;
import com.ztspeech.AppAudio;

import org.dragonnova.meetingapp.utils.Constants;
import org.dragonnova.meetingapp.utils.LogUtil;
import org.json.JSONException;
import org.json.JSONObject;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Locale;

/**
 * Created by hucn on 2016/4/11.
 * Description:科大讯飞的语音识别类
 */
public class XFRecognizer {

    private final static String TAG = XFRecognizer.class.getName();

    private SpeechRecognizer mIat;

    private SharedPreferences mSharedPreferences;
    // 引擎类型
    private String mEngineType = SpeechConstant.TYPE_CLOUD;
    // 打印类
    private Toast mToast;
    // 用HashMap存储听写结果
    private HashMap<String, String> mIatResults = new LinkedHashMap<String, String>();

    private int ret = 0; // 函数调用返回值

    private Context mContext;

    private Handler mHandler;

    private static final String AUDIO_RECORDER_FOLDER = "org.dragonnova.meetingapp/AudioRecorder";

    private String mVoiceSavedDirectory;

    private String mNewVoicePath;
    //设置是否连续识别
    private boolean isContinuous = false;
    //传递给handler的msg信息
    public static int continuousMsg = 1563645;

    private final String CH = "zh_cn";
    private final String EN = "en_us";

    private static final boolean allowTipsShown = false;

    @SuppressLint("ShowToast")
    public XFRecognizer(Handler hanlder, Context context) {
        this.mHandler = hanlder;
        this.mContext = context;

        mSharedPreferences = context.getSharedPreferences(Constants.APP_GLOBAL_CONFIG,
                Context.MODE_PRIVATE);

        // 将“12345678”替换成您申请的APPID，申请地址：http://open.voicecloud.cn
        SpeechUtility.createUtility(context, SpeechConstant.APPID + "=570cf5a2");
        //1.创建SpeechRecognizer对象，第二个参数：本地听写时传InitListener
        mIat = SpeechRecognizer.createRecognizer(context, null);

        //创建保存录音文件的文件夹
        mVoiceSavedDirectory = Environment.getExternalStorageDirectory().getAbsolutePath() + "/" + AUDIO_RECORDER_FOLDER;

        //2.设置听写参数，详见《科大讯飞MSC API手册(Android)》SpeechConstant类
        setDefaultParam();

        mToast = Toast.makeText(mContext, "", Toast.LENGTH_SHORT);
    }

    /**
     * 默认参数设置
     */
    public void setDefaultParam() {
        // 清空参数
        mIat.setParameter(SpeechConstant.PARAMS, null);

        // 设置听写引擎
        mIat.setParameter(SpeechConstant.ENGINE_TYPE, mEngineType);
        // 设置返回结果格式
        mIat.setParameter(SpeechConstant.RESULT_TYPE, "json");

        String lag = mSharedPreferences.getString(Constants.APP_GLOBAL_CONFIG,
                "mandarin");
        if (lag.equals(EN)) {
            // 设置语言
            mIat.setParameter(SpeechConstant.LANGUAGE, EN);
        } else {
            // 设置语言
            mIat.setParameter(SpeechConstant.LANGUAGE, CH);
            // 设置语言区域
            mIat.setParameter(SpeechConstant.ACCENT, lag);
        }


        // 设置语音前端点:静音超时时间，即用户多长时间不说话则当做超时处理
        mIat.setParameter(SpeechConstant.VAD_BOS, mSharedPreferences.getString("iat_vadbos_preference", "4000"));

        // 设置语音后端点:后端点静音检测时间，即用户停止说话多长时间内即认为不再输入， 自动停止录音
        mIat.setParameter(SpeechConstant.VAD_EOS, mSharedPreferences.getString("iat_vadeos_preference", "1000"));
        //mIat.setParameter(SpeechConstant.KEY_SPEECH_TIMEOUT, "60000");

        // 设置标点符号,设置为"0"返回结果无标点,设置为"1"返回结果有标点
        mIat.setParameter(SpeechConstant.ASR_PTT, mSharedPreferences.getString("iat_punc_preference", "1"));

        setDefaultSavedParam();
    }

    /**
     * 设置语言参数
     */
    private void setLanguageParam(Boolean isCH2EN) {
        String isch2en = isCH2EN ? CH : EN;
        if (isch2en.equals(EN)) {
            mIat.setParameter(SpeechConstant.LANGUAGE, EN);
        } else {
            mIat.setParameter(SpeechConstant.LANGUAGE, CH);
        }
    }

    /**
     * 设置保存文件参数
     */
    private void setSavedParam(String file) {
        // 设置音频保存路径，保存音频格式支持pcm、wav，设置路径为sd卡请注意WRITE_EXTERNAL_STORAGE权限
        mNewVoicePath = mVoiceSavedDirectory + "/" + file;
        // 注：AUDIO_FORMAT参数语记需要更新版本才能生效
        mIat.setParameter(SpeechConstant.AUDIO_FORMAT, "wav");
        mIat.setParameter(SpeechConstant.ASR_AUDIO_PATH, mNewVoicePath);
    }

    /**
     * 设置默认保存文件
     */
    private void setDefaultSavedParam() {
        // 设置音频保存路径，保存音频格式支持pcm、wav，设置路径为sd卡请注意WRITE_EXTERNAL_STORAGE权限
        // 注：AUDIO_FORMAT参数语记需要更新版本才能生效
        mIat.setParameter(SpeechConstant.AUDIO_FORMAT, "wav");
        mIat.setParameter(SpeechConstant.ASR_AUDIO_PATH, mVoiceSavedDirectory + "/msc/iat.wav");
    }

    /**
     * 生成Wav文件的路径
     */
    private String getNewVoiceName() {
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMddHHmmss", Locale.CHINA);
        Date curDate = new Date(System.currentTimeMillis());
        return dateFormat.format(curDate) + ".wav";
    }

    /**
     * 设置是否连续录音
     */
    public void setContinuous(boolean isContinuous) {
        this.isContinuous = isContinuous;
    }

    /**
     * 听写监听器
     */
    private RecognizerListener recognizerListener = new RecognizerListener() {

        //开始录音
        @Override
        public void onBeginOfSpeech() {
            // 此回调表示：sdk内部录音机已经准备好了，用户可以开始语音输入
            showTip("开始说话");
        }

        //听写结果回调接口(返回Json格式结果，用户可参见附录12.1)；
        //一般情况下会通过onResults接口多次返回结果，完整的识别内容是多次结果的累加；
        //关于解析Json的代码可参见MscDemo中JsonParser类；
        //isLast等于true时会话结束。
        @Override
        public void onResult(RecognizerResult results, boolean isLast) {
            LogUtil.eee("Result:" + results.getResultString());
            String tempResult = getResult(results);
            Message msg = mHandler.obtainMessage(AppAudio.RECOGNIZE_TEMP_RESULT);
            Bundle bundle = new Bundle();
            bundle.putBoolean("isCH2EN", mIat.getParameter(SpeechConstant.LANGUAGE).equals(CH));
            bundle.putString("tempResult", tempResult);
            msg.setData(bundle);
            mHandler.sendMessage(msg);

            if (isLast) {
                // TODO 最后的结果
                Message msg1 = mHandler.obtainMessage(AppAudio.RECOGNIZED);
                bundle.putString("newFileName", mNewVoicePath);
                msg1.setData(bundle);
                if (isContinuous) {
                    msg1.arg1 = continuousMsg;
                }
                mHandler.sendMessage(msg1);
            }
        }

        //会话发生错误回调接口
        @Override
        public void onError(SpeechError error) {
            error.getPlainDescription(true); //获取错误码描述
            // Tips：
            // 错误码：10118(您没有说话)，可能是录音机权限被禁，需要提示用户打开应用的录音权限。
            // 如果使用本地功能（语记）需要提示用户开启语记的录音权限。
            showTip(error.getPlainDescription(true));
            Message msg1 = mHandler.obtainMessage(AppAudio.RECOGNIZE_ERROR);
            if (isContinuous) {
                msg1.arg1 = continuousMsg;
            }
            mHandler.sendMessage(msg1);
        }

        //音量值0~30
        @Override
        public void onVolumeChanged(int volume, byte[] data) {
            showTip("当前正在说话，音量大小：" + volume);
            LogUtil.d(TAG, "返回音频数据：" + data.length);
        }

        //结束录音
        @Override
        public void onEndOfSpeech() {
            // 此回调表示：检测到了语音的尾端点，已经进入识别过程，不再接受语音输入
            showTip("结束说话");
            Message msg1 = mHandler.obtainMessage(AppAudio.RECOGNIZED_FINISH);
            mHandler.sendMessage(msg1);
        }

        //扩展用接口
        @Override
        public void onEvent(int eventType, int arg1, int arg2, Bundle obj) {
            // 以下代码用于获取与云端的会话id，当业务出错时将会话id提供给技术支持人员，可用于查询会话日志，定位出错原因
            // 若使用本地能力，会话id为null
            //	if (SpeechEvent.EVENT_SESSION_ID == eventType) {
            //		String sid = obj.getString(SpeechEvent.KEY_EVENT_SESSION_ID);
            //		Log.d(TAG, "session id =" + sid);
            //	}
        }
    };

    private void showTip(final String str) {
        if (allowTipsShown) {
            mToast.setText(str);
            mToast.show();
        }
    }

    /**
     * 获取识别的结果
     */
    private String getResult(RecognizerResult results) {
        String text = JsonParser.parseIatResult(results.getResultString());

        String sn = null;
        // 读取json结果中的sn字段
        try {
            JSONObject resultJson = new JSONObject(results.getResultString());
            sn = resultJson.optString("sn");
        } catch (JSONException e) {
            e.printStackTrace();
        }

        mIatResults.put(sn, text);

        StringBuilder resultBuffer = new StringBuilder();
        for (String key : mIatResults.keySet()) {
            resultBuffer.append(mIatResults.get(key));
        }
        return resultBuffer.toString();
    }

    /**
     * 语音识别，直接语音识别
     */
    private void recognize(boolean isCH2EN) {

        setLanguageParam(isCH2EN);
        //setSavedParam(getNewVoiceName());
        setDefaultSavedParam();
        mIatResults.clear();
        //开始听写
        int ret = mIat.startListening(recognizerListener);
        if (ret != ErrorCode.SUCCESS) {
            showTip("听写失败,错误码：" + ret);
        } else {
            showTip("请开始说话…");
        }
    }

    /**
     * 语音识别，文件识别
     */
    private void recognize(boolean isCH2EN, String resourceFile) {
        setLanguageParam(isCH2EN);
        setDefaultSavedParam();
        mIatResults.clear();
        // 设置音频来源为外部文件
        mIat.setParameter(SpeechConstant.AUDIO_SOURCE, "-1");
        // 也可以像以下这样直接设置音频文件路径识别（要求设置文件在sdcard上的全路径）：
        // mIat.setParameter(SpeechConstant.AUDIO_SOURCE, "-2");
        // mIat.setParameter(SpeechConstant.ASR_SOURCE_PATH, "sdcard/XXX/XXX.pcm");
        ret = mIat.startListening(recognizerListener);
        if (ret != ErrorCode.SUCCESS) {
            showTip("识别失败,错误码：" + ret);
        } else {
            byte[] audioData = FucUtil.readAudioFile(mContext, resourceFile);

            if (null != audioData) {
                showTip("正在识别...");
                // 一次（也可以分多次）写入音频文件数据，数据格式必须是采样率为8KHz或16KHz（本地识别只支持16K采样率，云端都支持），位长16bit，单声道的wav或者pcm
                // 写入8KHz采样的音频时，必须先调用setParameter(SpeechConstant.SAMPLE_RATE, "8000")设置正确的采样率
                // 注：当音频过长，静音部分时长超过VAD_EOS将导致静音后面部分不能识别
                mIat.writeAudio(audioData, 0, audioData.length);
                mIat.stopListening();
            } else {
                mIat.cancel();
                showTip("读取音频流失败");
            }
        }
    }

    /**
     * 语音识别，byte[] 识别
     */
    private void recognize(boolean isCH2EN, byte[] resourceBytes) {
        setLanguageParam(isCH2EN);
        setDefaultSavedParam();
        mIatResults.clear();
        mIat.setParameter(SpeechConstant.AUDIO_SOURCE, "-1");
        ret = mIat.startListening(recognizerListener);
        if (ret != ErrorCode.SUCCESS) {
            showTip("识别失败,错误码：" + ret);
        } else {
            if (null != resourceBytes) {
                showTip("正在识别...");
                mIat.writeAudio(resourceBytes, 0, resourceBytes.length);
                mIat.stopListening();
            } else {
                mIat.cancel();
                showTip("读取音频流失败");
            }
        }
    }

    /**
     * 开始识别,通过语音输入
     */
    public void start(Boolean isCH2EN) {
        recognize(isCH2EN);
    }

    /**
     * 开始识别，通过文件
     */
    public void start(Boolean isCH2EN, String voiceFile) {
        recognize(isCH2EN, voiceFile);
    }

    /**
     * 开始识别，通过bytes
     */
    public void start(Boolean isCH2EN, byte[] voiceBytes) {
        recognize(isCH2EN, voiceBytes);
    }

    /**
     * 停止且识别
     */
    public void stop() {
        mIat.stopListening();
        showTip("停止听写");
    }

    /**
     * 停止不识别
     */
    public void cancel() {
        mIat.cancel();
        showTip("取消听写");
    }
}
